import unittest
from c_16.country_codes import get_country_code

class TestCountryCodes(unittest.TestCase):
    def test_country_codes(self):
        country = 'Bolivia'
        code = get_country_code(country)
        self.assertEqual(code,'bo')

if __name__=="__main__":
	unittest.main()